import os


# 计算目录大小
def get_dir_size(p_dir):
    size = 0
    # 枚举子目录及目录下文件
    for file in os.listdir(p_dir):
        fullname = os.path.join(p_dir, file)
        # 文件的场合--计算文件大小
        if os.path.isfile(fullname):
            size = size + os.path.getsize(fullname)
        # 目录的场合--递归寻找子目录文件
        if os.path.isdir(fullname):
            size = size + get_dir_size(fullname)
    return size


def main():
    root_dir = input(r'请输入文件夹路径（例:"D:\test"）:')
    print('Size : {0}Byte'.format(get_dir_size(root_dir)))


if __name__ == '__main__':
    main()
